# -*- coding:utf-8 -*-

import xlwt

'''
 3.保存数据到Excel表格
'''
def saveDataToExcel(datalist,excelpath):
    print('保存数据到excel表格.......')
    # 以utf-8编码创建一个Excel对象
    # style_compression=0 表示单元格是否压缩，是否允许改变excl表格的样式
    workbook = xlwt.Workbook(encoding='utf-8', style_compression=0)
    # 创建一个sheet表
    # cell_overwrite_ok=True 允许多次写入，如果是False，多次写入会发生异常
    worksheet = workbook.add_sheet('sheet1', cell_overwrite_ok=True)
    # 创建一个元组，用来指定表格数据的标题
    col = ('电影中文名', '电影外文名', '电影详情链接', '封面图片地址', '电影评分', '评价人数', '一句话影评', '主创团队')
    # 把标题循环添加到excel表格
    for i in range(0, len(col)):
        worksheet.write(0, i, col[i])  # 列名
    # 把爬取到的电影信息添加到Excel表格
    # 循环添加，长度为250 datalist的长度
    for i in range(0, len(datalist)):
        print('第%d条' % (i + 1))
        # 为每一部电影的信息分别添加到每一行的每一个单元格当中
        data = datalist[i]
        for j in range(len(data)):
            worksheet.write(i + 1, j, data[j])  # 数据
    # 保存表格
    workbook.save(excelpath)
